/* @flow */

import { remove } from 'shared/util'

/**
 * 该模块主要用于处理Vue实例上的$refs属性，向该属性中
 * 新增、更新、删除vnode对象对应的vue实例或者el dom元素
 */
export default {
  //向vnode所属的vue实例中的$refs属性中新增当前vnode对象对应的vue实例或者el dom对象
  create (_: any, vnode: VNodeWithData) {
    registerRef(vnode)
  },
  update (oldVnode: VNodeWithData, vnode: VNodeWithData) {
    if (oldVnode.data.ref !== vnode.data.ref) {
      registerRef(oldVnode, true)
      registerRef(vnode)
    }
  },
  //从vnode所属的vue实例的$refs属性中移除当前vnode对象对应的vue实例或者el dom对象
  destroy (vnode: VNodeWithData) {
    registerRef(vnode, true)
  }
}

/**
 * 参数vnode所属于一个vue实例，而该方法就是用于更新vue实例中的
 * $refs属性，向$refs属性中添加、更新、或者移除vnode对应的vue实例或者el dom元素
 * @param vnode
 * @param isRemoval
 */
export function registerRef (vnode: VNodeWithData, isRemoval: ?boolean) {
  const key = vnode.data.ref
  if (!key) return

  //context指vnode所所属的Vue实例
  const vm = vnode.context
  //componentInstance指vnode所创建的Vue实例
  const ref = vnode.componentInstance || vnode.elm
  const refs = vm.$refs
  if (isRemoval) {
    if (Array.isArray(refs[key])) {
      //remove方法用于从一个数组中删除指定的数据项
      remove(refs[key], ref)
    } else if (refs[key] === ref) {
      refs[key] = undefined
    }
  } else {
    if (vnode.data.refInFor) {
      if (!Array.isArray(refs[key])) {
        refs[key] = [ref]
      } else if (refs[key].indexOf(ref) < 0) {
        // $flow-disable-line
        refs[key].push(ref)
      }
    } else {
      refs[key] = ref
    }
  }
}
